/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package aiaudio.database.tablegroup;

import aiaudio.lastfm.hbase.CannotCreateTableException;
import aiaudio.database.DatabaseStorage;
import aiaudio.database.tables.TableStructure;
import java.io.IOException;

/**
 *
 * @author nastya
 */
public class SplitDataSetAlgTableGroup extends TableGroup {

    private String trainingRatingMatrix = "training_set";

    public SplitDataSetAlgTableGroup(String key, boolean dropIfExists) {
        super(key, dropIfExists);
    }

    public void setTrainingRatingMatrix(String trainingRatingMatrix) {
        this.trainingRatingMatrix = trainingRatingMatrix;
    }

    public String getTrainingRatingMatrix() {
        return addKey(trainingRatingMatrix);
    }

    @Override
    public void create(DatabaseStorage storage) throws CannotCreateTableException, IOException {
        createTable(storage, getTrainingRatingMatrix(), TableStructure.RatingMatrix);
    }
}
